import requests
import app


# 创建类
class IhrmEmpManageApi:
    # 定义属性
    def __init__(self):
        self.url = app.BASE_URL + 'api/sys/user'

    # 定义方法 -- 登录成功 -- 写token 到全局变量
    def login_token(self, reqData):
        resp = requests.post(url=app.BASE_URL + 'api/sys/login', headers=app.HEADERS, json=reqData)

        app.TOKEN = "Bearer " + resp.json().get('data')  # 写入全局变量
        # 修改 现有的 headers， 添加新的 TOKEN,
        # 此行代码执行结束：app.py 中的 HEADERS就 变为 {"Content-Type": "application/json", "Authorization":"Bearer 2ed7d932-b3c2-4f38-ba75-15c395bc7583"}
        app.HEADERS['Authorization'] = app.TOKEN

    # 添加员工
    def add_emp(self, reqData):
        resp = requests.post(url=self.url, headers=app.HEADERS, json=reqData)
        return resp

    # 查询员工
    def query_emp(self, emp_id):
        return requests.get(url=app.BASE_URL + 'api/sys/user/' + emp_id, headers=app.HEADERS)

    # 修改员工
    def modify_emp(self, emp_id, reqData):
        return requests.put(url=app.BASE_URL + 'api/sys/user/' + emp_id,
                            headers=app.HEADERS, json=reqData)

    # 删除员工
    def delete_emp(self, emp_id):
        return requests.delete(url=app.BASE_URL + 'api/sys/user/' + emp_id, headers=app.HEADERS)

    # 查询员工列表
    def query_emp_list(self):
        return requests.get(url=app.BASE_URL + 'api/sys/user?page=1&size=10',headers=app.HEADERS)